package com.dh.leetcode.k1;


import org.junit.Assert;
import org.junit.Test;

/**
 * ClassName: _1518_water_bottles
 *
 * @author shouzimu
 * @Description: 1518. 换酒问题
 * 小区便利店正在促销，用 numExchange 个空酒瓶可以兑换一瓶新酒。你购入了 numBottles 瓶酒。
 * <p>
 * 如果喝掉了酒瓶中的酒，那么酒瓶就会变成空的。
 * <p>
 * 请你计算 最多 能喝到多少瓶酒。
 * <p>
 * 输入：numBottles = 9, numExchange = 3
 * 输出：13
 * 解释：你可以用 3 个空酒瓶兑换 1 瓶酒。
 * 所以最多能喝到 9 + 3 + 1 = 13 瓶酒。
 * <p>
 * <p>
 * 输入：numBottles = 15, numExchange = 4
 * 输出：19
 * 解释：你可以用 4 个空酒瓶兑换 1 瓶酒。
 * 所以最多能喝到 15 + 3 + 1 = 19 瓶酒。
 * 示例 3：
 * <p>
 * 输入：numBottles = 5, numExchange = 5
 * 输出：6
 * 示例 4：
 * <p>
 * 输入：numBottles = 2, numExchange = 3
 * 输出：2
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/water-bottles
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @date 2021/12/17 22:59
 */
public class _1518_water_bottles {

    public int numWaterBottles(int numBottles, int numExchange) {
        int sum = numBottles;
        while (numBottles >= numExchange) {
            sum += 1;
            numBottles -= numExchange;
            numBottles++;
        }
        return sum;
    }

    @Test
    public void numWaterBottlesTest() {
        Assert.assertEquals(19, numWaterBottles(15, 4));
        Assert.assertEquals(13, numWaterBottles(9, 3));
        Assert.assertEquals(6, numWaterBottles(5, 5));
        Assert.assertEquals(2, numWaterBottles(2, 3));
    }
}
